import numpy as np
arr2= np.array([[0,0,3],[1,5,6],[7,8,9]])
print(arr2)
v1_states = np.zeros_like(arr2)
v1_states[arr2 > 0] = 1

print(v1_states)
